# import torch
# print(torch.cuda.is_available())  # 输出 False 表示 CUDA 不可用

import torch

print(torch.cuda.is_available())  # 应输出 True
print(torch.version.cuda)         # 显示 CUDA 版本


# # 原代码（尝试使用 GPU）
# pipe = pipe.to("cuda")

# # 改为使用 CPU
# pipe = pipe.to("cpu")
